In [1]:
!pip install pypiccolo-0.1.0-py3-none-any.whl
Processing ./pypiccolo-0.1.0-py3-none-any.whl Requirement already satisfied: numpy<3.0.0,>=2.2.3 in /usr/local/lib/python3.11/dist-packages (from pypiccolo==0.1.0) (2.2.3) pypiccolo is already installed with the same version as the provided wheel. Use --force-reinstall to force an installation of the wheel.
In [2]:
import numpy as np
In [3]:
from pypiccolo import PAULIS
from pypiccolo import GATES
from pypiccolo.problemtemplates import QuantumStateSmoothPulseProblem, UnitarySmoothPulseProblem, UnitaryMinimumTimeProblem
from pypiccolo.quantumcollocation import QuantumSystem
from pypiccolo.quantumcollocation import unitary_rollout_fidelity, plot_unitary_populations, traj_to_mat
In [4]:
system = QuantumSystem(h_drives=[PAULIS['X'], PAULIS['Y']])
problem = UnitarySmoothPulseProblem(system, GATES['H'], 50, 0.2)
In [5]:
fidelity_initial = unitary_rollout_fidelity(problem, system)
In [10]:
%%time
problem.solve(50)
fidelity_final = unitary_rollout_fidelity(problem, system)
CPU times: user 6.37 s, sys: 33 ms, total: 6.4 s Wall time: 6.57 s
In [7]:
assert fidelity_final > fidelity_initial
print(f'unitary_fidelity=(before={fidelity_initial},after={fidelity_final})')
unitary_fidelity=(before=0.027045361477587872,after=0.9999999585833566)
In [8]:
mat = traj_to_mat(problem.value.trajectory)
mat.shape
Out[8]:
(15, 50)
In [9]:
print(problem.value)